import os

from paddleocr import PaddleOCR


def Img_path():
    # 文件夹路径
    folder_path = 'Img'

    # 遍历文件夹中的文件
    for filename in os.listdir(folder_path):
        # 检查文件是否为图片文件（这里简单地通过后缀名进行检查）
        if filename.endswith(('.png', '.jpg', '.jpeg', '.gif')):
            img_path = os.path.join(folder_path, filename)
            ORC(img_path)
        else:
            # 如果不是图片文件，可以选择跳过
            continue


def ORC(img_path):
    ocr = PaddleOCR(use_angle_cls=True, lang='ch')  # 创建PaddleOCR对象，use_angle_cls参数表示是否使用角度分类器，lang参数表示语言（中文）
    result = ocr.ocr(img_path, cls=True)  # 调用PaddleOCR对象的ocr方法进行图片识别，cls参数表示是否进行字符分类
    for line in result[0]:  # 遍历识别结果中的每一行也就是诗名
        print(line)  # 打印每一行的识别结果
    with open(f'Text/{result[0][0][1][0]}.txt', '+w', encoding='utf-8') as f:  # 打开识别结果中的第一行的第一个元素对应的文件，并以写入模式打开
        for line in result[0]:  # 遍历识别结果中的每一行
            f.write(line[1][0] + '\n')  # 将每一行的第二个元素的第一个字符写入文件，并换行


if __name__ == '__main__':
    Img_path()
